package com.cy.deepseeksport.NLP;

import io.github.pigmesh.ai.deepseek.core.DeepSeekClient;
import org.springframework.context.annotation.Bean;
import org.springframework.stereotype.Component;

@Component
public class DeepseekConfig {

    @Bean
    public DeepSeekClient huoshanDeepSeekClient() {

        DeepSeekClient.Builder builder = DeepSeekClient.builder()
                .baseUrl("https://ark.cn-beijing.volces.com/api/v3/bots")
                .model("bot-20250527143515-9kkhv")
                .openAiApiKey("26f9ab1f-3509-4f8e-b06e-7495bc93373a")
                .logRequests(true)
                .logResponses(true);

        return builder.build();
    }

    @Bean
    public DeepSeekClient oupaiDeepSeekClient() {

        DeepSeekClient.Builder builder = DeepSeekClient.builder()
                .baseUrl("https://api.ppinfra.com/v3/openai")
                .model("meta-llama/llama-3.3-70b-instruct")
                .openAiApiKey("sk_jfm4pv8rMcyCcOGjjvIpq4kuR9jQHvW5D4WrGcW6zX8")
                .logRequests(true)
                .logResponses(true);

        return builder.build();
    }

    @Bean
    public DeepSeekClient deepSeekr1Client() {

        DeepSeekClient.Builder builder = DeepSeekClient.builder()
                .baseUrl("https://api.ppinfra.com/v3/openai")
                .model("deepseek/deepseek-r1-turbo")
                .openAiApiKey("sk_jfm4pv8rMcyCcOGjjvIpq4kuR9jQHvW5D4WrGcW6zX8")
                .logRequests(true)
                .logResponses(true);

        return builder.build();
    }

    @Bean
    public DeepSeekClient qwenClient() {

        DeepSeekClient.Builder builder = DeepSeekClient.builder()
                .baseUrl("https://api.ppinfra.com/v3/openai")
                .model("qwen/qwen3-235b-a22b-fp8")
                .openAiApiKey("sk_jfm4pv8rMcyCcOGjjvIpq4kuR9jQHvW5D4WrGcW6zX8")
                .logRequests(true)
                .logResponses(true);

        return builder.build();
    }
}
